#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Create by zhang
# Create on 2022/7/8 14:53
from enum import IntEnum


class TransactionLabelEnum(IntEnum):
    sell = 1
    keep = 2
    buy = 3
    other = 4

    @classmethod
    def get_items(cls):
        return [cls.sell, cls.keep, cls.buy]

    @classmethod
    def get_names(cls):
        return [item.name for item in cls.get_items()]